package com.zhongwang.mms.util;

import com.alibaba.excel.EasyExcel;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.List;

public class ExcelExportUtil{
    /**
     * 导出 Excel 文件（通用方法）
     * @param data 导出的数据列表
     * @param clazz 实体类Class（如 User.class）
     * @param fileName 导出的文件名（不含 .xlsx 后缀）
     * @return ResponseEntity<byte[]> 用于 Spring MVC 响应下载
     */
    public static <T> ResponseEntity<byte[]> exportExcel(List<T> data, Class<T> clazz, String fileName) {
        try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
            // 代码与 3.x 版本完全一致
            EasyExcel.write(outputStream, clazz)
                    .sheet("sheet1")
                    .doWrite(data);

            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            // 解决中文文件名乱码
            headers.setContentDispositionFormData("attachment",
                    new String((fileName + ".xlsx").getBytes("UTF-8"), "ISO-8859-1"));

            return new ResponseEntity<>(outputStream.toByteArray(), headers, HttpStatus.OK);
        } catch (IOException e) {
            e.printStackTrace();
            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
        }
    }
}
